# Basic Dockerfile
FROM nvidia/cuda:11.7.1-base-ubuntu22.04

COPY cuda_mem_and_sm_count /tmp
CMD ["/tmp/cuda_mem_and_sm_count"]
